/*
 * Copyright (c) 2018-2999 广州亚米信息科技有限公司 All rights reserved.
 *
 * https://www.gz-yami.com/
 *
 * 未经允许，不可做商业用途！
 *
 * 版权所有，侵权必究！
 */

package com.yami.shop.api.controller;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.yami.shop.bean.app.dto.ProductDto;
import com.yami.shop.bean.app.dto.UserCollectionDto;
import com.yami.shop.bean.model.Product;
import com.yami.shop.bean.model.UserCollection;
import com.yami.shop.common.exception.YamiShopBindException;
import com.yami.shop.common.util.PageParam;
import com.yami.shop.security.util.SecurityUtils;
import com.yami.shop.service.ProductService;
import com.yami.shop.service.UserCollectionService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiImplicitParam;
import io.swagger.annotations.ApiOperation;
import lombok.AllArgsConstructor;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.*;

import java.util.Date;
import java.util.Objects;

@RestController
@RequestMapping("/p/user/collection")
@Api(tags = "收藏接口")
@AllArgsConstructor
public class UserCollectionController {

	private final UserCollectionService userCollectionService;

	private final ProductService productService;

	@SuppressWarnings("rawtypes")
	@GetMapping("/page")
	@ApiOperation(value = "分页返回收藏数据", notes = "根据用户id获取")
	public ResponseEntity<IPage<UserCollectionDto>> getUserCollectionDtoPageByUserId(PageParam page) {
		return ResponseEntity
				.ok(userCollectionService.getUserCollectionDtoPageByUserId(page, SecurityUtils.getUser().getUserId()));
	}

	@GetMapping("isCollection")
	@ApiOperation(value = "根据商品id获取该商品是否在收藏夹中", notes = "传入收藏商品id")
	public ResponseEntity<Boolean> isCollection(Long prodId) {
		if (productService.count(new LambdaQueryWrapper<Product>().eq(Product::getProdId, prodId)) < 1) {
			throw new YamiShopBindException("该商品不存在");
		}
		return ResponseEntity.ok(userCollectionService
				.count(new LambdaQueryWrapper<UserCollection>().eq(UserCollection::getProdId, prodId)
						.eq(UserCollection::getUserId, SecurityUtils.getUser().getUserId())) > 0);
	}

	@PostMapping("/addOrCancel")
	@ApiOperation(value = "添加/取消收藏", notes = "传入收藏商品id,如果商品未收藏则收藏商品，已收藏则取消收藏")
	@ApiImplicitParam(name = "prodId", value = "商品id", required = true, dataType = "Long")
	public ResponseEntity<Void> addOrCancel(@RequestBody Long prodId) {
		if (Objects.isNull(productService.getProductByProdId(prodId))) {
			throw new YamiShopBindException("该商品不存在");
		}
		String userId = SecurityUtils.getUser().getUserId();
		if (userCollectionService.count(new LambdaQueryWrapper<UserCollection>().eq(UserCollection::getProdId, prodId)
				.eq(UserCollection::getUserId, userId)) > 0) {
			userCollectionService.remove(new LambdaQueryWrapper<UserCollection>().eq(UserCollection::getProdId, prodId)
					.eq(UserCollection::getUserId, userId));
		} else {
			UserCollection userCollection = new UserCollection();
			userCollection.setCreateTime(new Date());
			userCollection.setUserId(userId);
			userCollection.setProdId(prodId);
			userCollectionService.save(userCollection);
		}
		return ResponseEntity.ok().build();
	}

	/**
	 * 查询用户收藏商品数量
	 */
	@GetMapping("count")
	@ApiOperation(value = "查询用户收藏商品数量", notes = "查询用户收藏商品数量")
	public int findUserCollectionCount() {
		String userId = SecurityUtils.getUser().getUserId();
		return userCollectionService
				.count(new LambdaQueryWrapper<UserCollection>().eq(UserCollection::getUserId, userId));
	}

	@SuppressWarnings("rawtypes")
	@GetMapping("/prods")
	@ApiOperation(value = "获取用户收藏商品列表", notes = "获取用户收藏商品列表")
	public ResponseEntity<IPage<ProductDto>> collectionProds(PageParam page) {
		String userId = SecurityUtils.getUser().getUserId();
		IPage<ProductDto> productDtoIPage = productService.collectionProds(page, userId);
		return ResponseEntity.ok(productDtoIPage);
	}

}
